# Mission 10 Battery Tester
# Objective 4 - final code
from botcore import *
from time import sleep

def vbatt_load():
    # Turn on 4 LEDs & read volts
    leds.user(15)
    v = system.pwr_volts()
    leds.user(0)
    return v

def batt_level(v):
    pct = (v / 2) - 2

    if pct > 1:
        pct = 1.0
    elif pct < 0:
        pct = 0.0

    print("Batt Level:", v, "->", pct)
    return pct

def check_batt(pct):
    if pct < 0.2:
        leds.pwr(True)
        return

    if pct > 0.6:
        blinks = 1
    else:
        blinks = 2

    while blinks > 0:
        leds.pwr(True)
        sleep(0.5)
        leds.pwr(False)
        sleep(0.2)
        blinks = blinks - 1

# -- Main program --
print("Current load")
vb = vbatt_load()
my_capacity = batt_level(vb)
check_batt(my_capacity)

# -- Test data --
print("Test data")
vb = 5.8
my_capacity = batt_level(vb)
check_batt(my_capacity)